import torch


class CnnStockConfig:
    batch_size = 16
    learning_rate = 0.005
    num_epochs = 2000
    input_dim = 15
    num_classes = 9
    early_stop_patience = 2000
    ratio = 0.8
    device = torch.device('cpu')
